package dao

import (
	"fmt"
	"go_web_bubble/setting"
	"log"
	"os"
	"time"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

var (
	a1 int
	DB *gorm.DB
)

//待优化用户名和密码等以配置方式传递
func InitMysql(cfg *setting.MySQLConfig) (err error) {

	// 新版本
	slowLogger := logger.New(
		//将标准输出作为Writer
		log.New(os.Stdout, "\r\n", log.LstdFlags),
		logger.Config{
			//设定慢查询时间阈值为1ms
			SlowThreshold: 1 * time.Microsecond,
			//设置日志级别，只有Warn和Info级别会输出慢查询日志
			LogLevel: logger.Warn,
		},
	)
	//dsn := "root:5120154230@(localhost)/bubble?charset=utf8mb4&parseTime=True&loc=Local"
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DataBase)
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{ //建立连接时指定打印info级别的sql
		//Logger: logger.Default.LogMode(logger.Info), //配置日志级别，打印出所有info的sql
		Logger: slowLogger,
	})
	if err != nil {
		return
		//panic(err)
	}
	sqlDB, _ := DB.DB()
	return sqlDB.Ping()
}
func Close() {
	sqlDB, err := DB.DB()
	if err != nil {
		panic(err)
	}
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(10)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)

	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
	sqlDB.Close()
}
